/////////////////////////////////////////////////////////////////////////////
// Original code from libhdfs3. Copyright (c) 2013 - 2014, Pivotal Inc.
// All rights reserved. Author: Zhanwei Wang
/////////////////////////////////////////////////////////////////////////////
//  Modifications by Kumo Inc.
// Copyright (C) Kumo inc. and its affiliates.
// Author: Jeff.li lijippy@163.com
// All rights reserved.
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
//



#include <kmhdfs/client/file_system_impl.h>
#include <kmhdfs/client/file_system_inter.h>
#include <kmhdfs/client/input_stream.h>
#include <kmhdfs/client/input_stream_impl.h>
#include <kmhdfs/client/input_stream_inter.h>
#include <kmhdfs/client/striped_input_stream_impl.h>

using namespace Hdfs::Internal;

namespace Hdfs {

InputStream::InputStream() {
    impl = new Internal::InputStreamImpl;
}

InputStream::InputStream(std::shared_ptr<LocatedBlocks> lbs) {
    if (lbs->getEcPolicy()) {
        impl = new Internal::StripedInputStreamImpl(lbs);
    } else {
        impl = new Internal::InputStreamImpl(lbs);
    }
}

InputStream::~InputStream() {
    delete impl;
}

/**
 * Open a file to read
 * @param fs hdfs file system.
 * @param path the file to be read.
 * @param verifyChecksum verify the checksum.
 */
void InputStream::open(FileSystem & fs, const char * path,
                       bool verifyChecksum) {
    if (!fs.impl) {
        THROW(HdfsIOException, "FileSystem: not connected.");
    }

    impl->open(fs.impl->filesystem, path, verifyChecksum);
}

/**
 * To read data from hdfs.
 * @param buf the buffer used to filled.
 * @param size buffer size.
 * @return return the number of bytes filled in the buffer, it may less than size.
 */
int32_t InputStream::read(char * buf, int32_t size) {
    return impl->read(buf, size);
}

/**
 * To read data from hdfs.
 * @param buf the buffer used to filled.
 * @param size buffer size.
 * @param position the position to seek.
 * @return return the number of bytes filled in the buffer, it may less than size.
 */
int32_t InputStream::pread(char * buf, int32_t size, int64_t position) {
    return impl->pread(buf, size, position);
}

/**
 * To read data from hdfs, block until get the given size of bytes.
 * @param buf the buffer used to filled.
 * @param size the number of bytes to be read.
 */
void InputStream::readFully(char * buf, int64_t size) {
    impl->readFully(buf, size);
}

int64_t InputStream::available() {
    return impl->available();
}

/**
 * To move the file point to the given position.
 * @param pos the given position.
 */
void InputStream::seek(int64_t pos) {
    impl->seek(pos);
}

/**
 * To get the current file point position.
 * @return the position of current file point.
 */
int64_t InputStream::tell() {
    return impl->tell();
}

/**
 * Close the sthream.
 */
void InputStream::close() {
    impl->close();
}

}
